import os
import pandas  as pd
from pyecharts import options as opts
from pyecharts.charts import Map

#定义数据目录
data_dir = ''
#定义背景图片
background_image = ''
#定义输出图表存放位置（html格式导出）
output_path = ''

#读取用户访问量类型数据
#假设数据是以省份为索引的一堆数组，或可以根据实际数据格式调整
data_file = os.path.join(data_dir,'user_access_data.csv')
data = pd.read_csv(data_file)

#假设数据中有两个列，‘province’和‘access_count’，分别表示省份和访问量
province_data = data[['province','access_count']].values.tolist()

#创建地图
map_chart = (
    Map()
    .add("用户访问量", province_data, "china")
    .set_global_opts(
        title_opts=opts.TitleOpts(title="不同省份用户访问量分布图"),
        visualmap_opts=opts.VisualMapOpts(max_=max(data['access_count']), is_piecewise=True),
    )
)

#设置背景图片
map_chart.render_background_img(background_image)
#生成图表
map_chart.render(output_path)
#导出
print(f"图表已生成并保存到: {output_path}")

